diff --git a/main.go b/main.go
index 6efbba5..f8ef586 100644
--- a/main.go
+++ b/main.go
@@ -4,14 +4,10 @@ import (
     "log"
     "net/http"
     "yaagobackend/routes" // Import local package (assuming router.go is in a subdirectory named 'routes')
-    "github.com/gorilla/mux"
 )
 
 func main() {
-    r := mux.NewRouter()
-
-    // Initialize routes from router.go
-    routes.InitializeRoutes(r)
+ 	r :=  routes.InitializeRouter()
 
     // Start the server
     log.Println("Server listening on port 8000")
diff --git a/routes/router.go b/routes/router.go
index 2f5046b..713b9bc 100644
--- a/routes/router.go
+++ b/routes/router.go
@@ -5,14 +5,14 @@ import (
     "yaagobackend/controllers" // Import local package (assuming router.go is in a subdirectory named 'routes')
 )
 
-// InitializeRouter sets up the router with all the routes
+// InitializeRouter sets up the router with all the routes and returns the router
 func InitializeRouter() *mux.Router {
-	r := mux.NewRouter()
+    r := mux.NewRouter()
 
-	// Define routes and their handlers
-	r.HandleFunc("/users", GetUsers).Methods("GET")
-	r.HandleFunc("/users/{id}", GetUser).Methods("GET")
-	r.HandleFunc("/users", CreateUser).Methods("POST")
+    // Define routes and their handlers
+    r.HandleFunc("/users", controllers.GetUsers).Methods("GET")
+    r.HandleFunc("/users/{id}", controllers.GetUser).Methods("GET")
+    r.HandleFunc("/users", controllers.CreateUser).Methods("POST")
 
-	return r
-}
+    return r
+}
\ No newline at end of file