1 files changed, 9 insertions, 4 deletions

diff --git a/api/cmd/main.go b/api/cmd/main.go
index d97c942..c23eff1 100644
--- a/api/cmd/main.go
+++ b/api/cmd/main.go
@@ -1,16 +1,21 @@
 package main
 
 import (
-        "water/api/internal/database"
+        "log"
+        "water/api/internal/config"
         "water/api/internal/router"
 )
 
 func main() {
-        database.SetupDatabase()
+        c, err := config.Load()
+        if err != nil {
+                log.Fatalf("Error while reading config file %s", err)
+        }
+
         r := router.SetupRouter()
         // Listen and Server in 0.0.0.0:8080
-        err := r.Run(":8080")
+        err = r.Run(c.GetString("PORT"))
         if err != nil {
-                return
+                log.Fatal(err)
         }
 }