diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index a236362d5a99c906828890e8252d14d96ea8c8f1..8969af3ef989b652fce96045a8fc407780e384c3 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -1041,7 +1041,10 @@ class RailEnv(Environment):
                 nbits = 0
                 tmp = self.rail.get_transitions((pos[0], pos[1]))
                 possible_transitions = self.rail.get_transitions((pos[0], pos[1], direction))
-                # print(np.sum(self.rail.get_transitions((pos[0], pos[1],direction))),self.rail.get_transitions((pos[0], pos[1],direction)),self.rail.get_transitions((pos[0], pos[1])),(pos[0], pos[1],direction))
+                # print(np.sum(self.rail.get_transitions((pos[0], pos[1],direction))),
+                # self.rail.get_transitions((pos[0], pos[1],direction)),
+                # self.rail.get_transitions((pos[0], pos[1])),
+                # (pos[0], pos[1],direction))
 
                 while tmp > 0:
                     nbits += (tmp & 1)
@@ -1087,7 +1090,7 @@ class RailEnv(Environment):
                     if np.sum(possible_transitions) == 1:
                         # Checking for curves
                         curv_dir = np.argmax(possible_transitions)
-                        #valid_transition = self.rail.get_transition(
+                        # valid_transition = self.rail.get_transition(
                         #    (pos[0], pos[1], direction),
                         #    movement)
                         movement = curv_dir