diff --git a/src/controllers/models/setup.go b/src/controllers/models/setup.go
index f4ed467..d982fbd 100644
--- a/src/controllers/models/setup.go
+++ b/src/controllers/models/setup.go
@@ -7,8 +7,8 @@ import (
 
 var DB *gorm.DB
 
-func ConnectDatabase() {
-	database, err := gorm.Open("sqlite3", "../test.db")
+func ConnectDatabase(filename *string) {
+	database, err := gorm.Open("sqlite3", *filename)
 
 	if err != nil {
 		panic("Failed to connect to database!")
diff --git a/src/main.go b/src/main.go
index 10c5247..af2136a 100644
--- a/src/main.go
+++ b/src/main.go
@@ -1,15 +1,20 @@
 package main
 
 import (
+	"flag"
+
 	"github.com/gin-gonic/gin"
 	"gitlab.celogeek.com/photos/api/src/controllers"
 	"gitlab.celogeek.com/photos/api/src/controllers/models"
 )
 
 func main() {
+	dbFilename := flag.String("db", "test.db", "database filename")
+	flag.Parse()
+
 	r := gin.Default()
 
-	models.ConnectDatabase()
+	models.ConnectDatabase(dbFilename)
 
 	r.GET("/", controllers.DefaultRoot)