diff --git a/main.py b/main.py
index c0205d2..fcde783 100644
--- a/main.py
+++ b/main.py
@@ -1,5 +1,5 @@
 import argparse
-import ArgumentParser
+from argparse import ArgumentParser
 from server import server
 
 def main():
@@ -27,4 +27,4 @@ def main():
         parser.print_help()
 
 if __name__ == "__main__":
-    exit(main())
\ No newline at end of file
+    exit(main())
diff --git a/run.sh b/run.sh
index d487422..0487431 100755
--- a/run.sh
+++ b/run.sh
@@ -7,4 +7,4 @@ cat ./Config/party.sql | sqlite3 party.db
 
 export FLASK_APP=main.py
 
-sudo python3 main.py
+sudo python3 main.py $@
diff --git a/shell.nix b/shell.nix
index c9f813e..08eeedb 100644
--- a/shell.nix
+++ b/shell.nix
@@ -14,4 +14,10 @@ pkgs.mkShell {
     pkgs.tmux
     pkgs.tree
   ];
+
+  shellHook = ''
+    export FLASK_APP=main.py
+    export FLASK_ENV=development
+    export FLASK_RUN_PORT=5000
+    '';
 }