diff --git a/main.go b/main.go
index 057e1d6..fc60c77 100644
--- a/main.go
+++ b/main.go
@@ -6,8 +6,9 @@ import (
 
 	_ "github.com/lib/pq"
 
-    . "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
-    . "git.andr3h3nriqu3s.com/andr3/fyp/logic/models"
+	. "git.andr3h3nriqu3s.com/andr3/fyp/logic/models"
+	models_utils "git.andr3h3nriqu3s.com/andr3/fyp/logic/models/utils"
+	. "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
 )
 
 const (
@@ -31,9 +32,13 @@ func main() {
 	fmt.Println("Starting server on :8000!")
 
     //TODO check if file structure exists to save data
-
 	handle := NewHandler(db)
 
+    _, err = db.Exec("update models set status=$1 where status=$2", models_utils.FAILED_TRAINING, models_utils.TRAINING);
+    if err != nil {
+        panic(err)
+    }
+
 
 	// TODO Handle this in other way
     handle.StaticFiles("/styles/", ".css", "text/css");