diff --git a/flatland/envs/agent_utils.py b/flatland/envs/agent_utils.py
index 6b881f9cda5d566b2989b938bb234d0560cdf7cb..2017d70610958241e1ddb8d35e8d88aac70d8686 100644
--- a/flatland/envs/agent_utils.py
+++ b/flatland/envs/agent_utils.py
@@ -38,13 +38,19 @@ class EnvAgentStatic(object):
                                 'speed': speeds[i] if speeds is not None else 1.0,
                                 'transition_action_on_cellexit': 0})
 
-        # TODO: on initialization, all agents are re-set as non-broken. Perhaps it may be desirable to set some as broken?
+        # TODO: on initialization, all agents are re-set as non-broken. Perhaps it may be desirable to set
+        # some as broken?
         broken_datas = []
         for i in range(len(positions)):
             broken_datas.append({'broken': 0,
                                 'number_of_halts': 0})
 
-        return list(starmap(EnvAgentStatic, zip(positions, directions, targets, [False] * len(positions), speed_datas, broken_datas)))
+        return list(starmap(EnvAgentStatic, zip(positions,
+                                                directions,
+                                                targets,
+                                                [False] * len(positions),
+                                                speed_datas,
+                                                broken_datas)))
 
     def to_list(self):
 
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index 79c86955819e6d353b4e31369d8304f4155b4357..0f03870bd141386d8ae4be1dd21b50e6d24c284f 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -167,7 +167,7 @@ class RailEnv(Environment):
 
         # Stochastic train breaking parameters
         self.min_average_broken_trains = 1
-        self.average_proportion_of_broken_trains = 0.1 # ~10% of the trains can be expected to break down in an episode
+        self.average_proportion_of_broken_trains = 0.1  # ~10% of the trains can be expected to break down in an episode
         self.mean_number_halts_to_break = 3
 
         # Uniform distribution