update

route/route.go

@@ -193,15 +193,16 @@ func Route(db *gorm.DB) {
 		w.Write([]byte("success"))
 	}))
 
-	http.HandleFunc("*", middleware.ResponseTimeMiddleware(func(w http.ResponseWriter, r *http.Request) {
-		w.WriteHeader(404)
-		w.Write([]byte("what are you looking for!?"))
-	}))
-
 	http.Handle("/", middleware.ResponseTimeMiddleware(func(w http.ResponseWriter, r *http.Request) {
 		if r.Method != http.MethodGet {
 			http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
 		}
-		w.Write([]byte("Welcome!"))
+
+		if r.URL.Path != "/" {
+			w.WriteHeader(404)
+			w.Write([]byte("What are you looking for?!"))
+		} else {
+			w.Write([]byte("Welcome!"))
+		}
 	}))
 }