diff --git a/.scripts/UpdateMoose.py b/.scripts/UpdateMoose.py
index 6a270bf..8d609cf 100644
--- a/.scripts/UpdateMoose.py
+++ b/.scripts/UpdateMoose.py
@@ -106,11 +106,13 @@ if __name__ == '__main__':
     parser = argparse.ArgumentParser(description='Blabla.')
 
     # Add argument for Moose path.
-    parser.add_argument('MoosePath', metavar='moose', type=str, help='path to Moose.lua file', default="./")
+    parser.add_argument('--MoosePath', metavar='moose', type=str, help='path to Moose.lua file', default="./")
 
     # Add argument for Moose path.
-    parser.add_argument('MissionPath', metavar='missions', type=str, help='path to missions', default="./")
+    parser.add_argument('--MissionPath', metavar='missions', type=str, help='path to missions', default="./")
 
+    #
+    parser.add_argument('--UpdateMoose', action='store_true')
 
     # Execute the parse_args() method
     args = parser.parse_args()
@@ -121,6 +123,12 @@ if __name__ == '__main__':
     # Moose.lua file
     MooseLua=Moose/"Moose_.lua"
 
+    if args.UpdateMoose:
+        print("Will update all Moose_.lua files")
+
+    print(args.MoosePath)
+    print(args.MissionPath)
+
     # Check that Moose.lua exists
     if MooseLua.exists():
         print("Moose_.lua exists")
diff --git a/.scripts/run.sh b/.scripts/run.sh
index a662982..cc2da48 100644
--- a/.scripts/run.sh
+++ b/.scripts/run.sh
@@ -1,3 +1,3 @@
 #!/bin/bash
 cd "$(dirname "$0")"
-python ./UpdateMoose.py ../../MOOSE_INCLUDE/Moose_Include_Static/ ../
\ No newline at end of file
+python ./UpdateMoose.py --MoosePath ../../MOOSE_INCLUDE/Moose_Include_Static/ --MissionPath ../ --UpdateMoose