diff --git a/aicrowd.json b/aicrowd.json
index 8f275f1d09091e94bcad7c9cde7d0acf4b2ab542..68c76af4fd222127604c5e5e3252429f9795fa4c 100644
--- a/aicrowd.json
+++ b/aicrowd.json
@@ -2,6 +2,6 @@
   "challenge_id": "neurips-2020-flatland-challenge",
   "grader_id": "neurips-2020-flatland-challenge",
   "debug": false,
-  "tags": ["other"]
+  "tags": ["RL"]
 }
 
diff --git a/run.py b/run.py
index a7c8f5b2c042dd719a91da4944e633a4daed924a..d3119e6bb08b38dd1ee627daf5bc27ed4e35f815 100644
--- a/run.py
+++ b/run.py
@@ -51,17 +51,17 @@ set_action_size_full()
 # Print per-step logs
 VERBOSE = True
 USE_FAST_TREEOBS = True
-USE_PPO_AGENT = True
-
-# Checkpoint to use (remember to push it!)
-# 116591 adrian_egli
-# graded	71.305	0.633	RL	Successfully Graded ! More details about this submission can be found at:
-# http://gitlab.aicrowd.com/adrian_egli/neurips2020-flatland-starter-kit/issues/51
-# Fri, 22 Jan 2021 23:37:56
-set_action_size_reduced()
-load_policy = "DDDQN"
-checkpoint = "./checkpoints/210122120236-3000.pth"  # 17.011131341978228
-EPSILON = 0.0
+
+if False:
+    # Checkpoint to use (remember to push it!)
+    # 116591 adrian_egli
+    # graded	71.305	0.633	RL	Successfully Graded ! More details about this submission can be found at:
+    # http://gitlab.aicrowd.com/adrian_egli/neurips2020-flatland-starter-kit/issues/51
+    # Fri, 22 Jan 2021 23:37:56
+    set_action_size_reduced()
+    load_policy = "DDDQN"
+    checkpoint = "./checkpoints/210122120236-3000.pth"  # 17.011131341978228
+    EPSILON = 0.0
 
 if False:
     # 116658 adrian_egli
@@ -83,12 +83,11 @@ if True:
     checkpoint = "./checkpoints/210122165109-5000.pth"  # 17.993750197899438
     EPSILON = 0.0
 
-
-if True:
-    # 116659 adrian_egli
-    # graded	80.579	0.715	RL	Successfully Graded ! More details about this submission can be found at:
-    # http://gitlab.aicrowd.com/adrian_egli/neurips2020-flatland-starter-kit/issues/53
-    # Sat, 23 Jan 2021 07:45:49
+if False:
+    # 116727 adrian_egli
+    # graded	106.786	0.768	RL	Successfully Graded ! More details about this submission can be found at:
+    # http://gitlab.aicrowd.com/adrian_egli/neurips2020-flatland-starter-kit/issues/54
+    # Sat, 23 Jan 2021 14:31:50
     set_action_size_reduced()
     load_policy = "DeadLockAvoidance"
     checkpoint = None