1 files changed, 4 insertions, 8 deletions
diff --git a/src/main.rs b/src/main.rs
index b1979b1..73dccef 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -9,12 +9,12 @@ use axum::{
     Router,
 };
 
-use routes::problem::{get_problems, create_problem};
+use routes::problem::{get_problem, get_problems, create_problem};
 use routes::user::{create_user, me};
 use routes::auth::{login, logout};
 use tower_http::services::ServeDir;
 
-use crate::{database::Database, utils::register_admin};
+use crate::{database::Database, routes::submission::create_submission, utils::register_admin};
 
 #[derive(Clone)]
 struct AppState {
@@ -61,8 +61,10 @@ async fn main() {
     let app = Router::new()
         .route("/login", post(login))
         .route("/logout", post(logout))
+        .route("/problem/{problem_id}", get(get_problem))
         .route("/problem", get(get_problems))
         .route("/problem", post(create_problem))
+        .route("/submission", post(create_submission))
         .route("/user", post(create_user))
         .route("/me", get(me))
         .nest_service("/static", static_files)
@@ -72,9 +74,3 @@ async fn main() {
     axum::serve(listener, app).await.unwrap();
 }
 
-#[cfg(test)]
-mod tests {
-    fn test_app() -> String {
-        "hello".into()
-    }
-}