diff --git a/run.py b/run.py
index e53f1e4..37548e8 100644
--- a/run.py
+++ b/run.py
@@ -14,12 +14,11 @@ problems = [
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
-    parser.add_argument('--problems', type=list,
+    parser.add_argument('-p', '--problem', type=str,
                         choices=problems,
-                        default=problems[-1:])
+                        default=problems[-1])
     args = parser.parse_args()
-    for problem in args.problems:
-        print(problem)
+    print(args.problem)
 
-        sys.path.append(os.path.join(os.getcwd(), problem))
-        __import__(problem, fromlist=['test'])
+    sys.path.append(os.path.join(os.getcwd(), args.problem))
+    __import__(args.problem, fromlist=['test'])