diff --git a/src/rbm/rbm.py b/src/rbm/rbm.py
index e099c9d..bf0f20a 100644
--- a/src/rbm/rbm.py
+++ b/src/rbm/rbm.py
@@ -1,6 +1,7 @@
 import os.path
 import numpy as np
 import cv2 as cv
+from argparse import ArgumentParser
 from stack_factory import StackFactory
 from status import Status
 from stack_deep import StackDeep
@@ -90,8 +91,15 @@ def main(prj_name: str = "test"):
 	stack.state_save()
 
 if __name__ == "__main__":
-#	main("norb_small_16h_v2")
-	main("mnist_2")
+	ap = ArgumentParser()
+	ap.add_argument("name", type=str,
+		default='default',
+		help="Name of project")
+
+	args = ap.parse_args()
+	var_args = vars(args)
+
+	main(var_args["name"])
 	cv.destroyAllWindows()
 
 	print("Test: [passed]")