diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 908df4322d58ec13729ce31cb2eebdb8ac0a74c2..5cc3f26d0374cb07a673d05771cbbf64a66f0fc0 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -266,7 +266,6 @@ class TreeObsForRailEnv(ObservationBuilder):
         other_agent_same_direction = 0
         other_agent_opposite_direction = 0
 
-
         num_steps = 1
         while exploring:
             # #############################
@@ -378,7 +377,6 @@ class TreeObsForRailEnv(ObservationBuilder):
                            ]
         """
 
-
         if last_isTarget:
             observation = [0,
                            other_target_encountered,
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index c98b1689f95b18e5a1baa2d57c270bc2fcae14cc..7694bb207a0610deb50777346ec4d00fcb55945a 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -215,22 +215,14 @@ class RailEnv(Environment):
 
             if action == RailEnvActions.DO_NOTHING and agent.moving:
                 # Keep moving
-                # Changed MOVE_FORWARD to DO_NOTHING
-                #action_dict[iAgent] = RailEnvActions.DO_NOTHING
                 action = RailEnvActions.MOVE_FORWARD
 
             if action == RailEnvActions.STOP_MOVING and agent.moving:
-                #action_dict[iAgent] = RailEnvActions.DO_NOTHING
-                # CHanged DO_NOTHING to STOP_MOVING
-                # action = RailEnvActions.STOP_MOVING
                 agent.moving = False
                 self.rewards_dict[iAgent] += stop_penalty
 
             if not agent.moving and action == RailEnvActions.MOVE_FORWARD:
                 # Only allow agent to start moving by pressing forward.
-                #(action == RailEnvActions.MOVE_LEFT or
-                # action == RailEnvActions.MOVE_FORWARD or
-                # action == RailEnvActions.MOVE_RIGHT):
                 agent.moving = True
                 self.rewards_dict[iAgent] += start_penalty