diff --git a/main.go b/main.go
index 1103ef6..d83536e 100644
--- a/main.go
+++ b/main.go
@@ -1,8 +1,9 @@
 package main
 
 import (
-	"freed/internal/api"
-	"freed/internal/database"
+	_ "embed"
+	"freed/api"
+	"freed/database"
 	"log"
 	"os"
 
@@ -18,7 +19,7 @@ func main() {
 		log.Fatalf("No ENV value set for 'DB_FILE', could not initialize database. Please provide a valid path and filename")
 	}
 
-	err := database.Connect(dbFile)
+	repository, err := database.NewRepository(dbFile)
 
 	if err != nil {
 		log.Fatalf("Could not initialize database: %v", err)
@@ -30,7 +31,7 @@ func main() {
 	app.Use(logger.New())
 
 	// Try to set up API routes
-	if err := api.Setup(app); err != nil {
+	if err := api.Setup(app, repository); err != nil {
 		log.Printf("Could not setup /api routes: %s", err)
 	}
 
diff --git a/sqlc.yaml b/sqlc.yaml
new file mode 100644
index 0000000..3703a15
--- /dev/null
+++ b/sqlc.yaml
@@ -0,0 +1,9 @@
+version: "2"
+sql:
+  - engine: "sqlite"
+    queries: "database/queries"
+    schema: "database/migrations"
+    gen:
+      go:
+        package: "database"
+        out: "database"