diff --git a/flatland/utils/graphics_layer.py b/flatland/utils/graphics_layer.py
index ec4641ff1c3c22a12d4da565f80d33e16237829c..d8d6d0d7938a0a084c1a996ff52db9ec59b54fdc 100644
--- a/flatland/utils/graphics_layer.py
+++ b/flatland/utils/graphics_layer.py
@@ -68,7 +68,7 @@ class GraphicsLayer(object):
     def get_cmap(self, *args, **kwargs):
         return plt.get_cmap(*args, **kwargs)
 
-    def setRailAt(self, row, col, binTrans, iTarget=None,isSelected=False):
+    def setRailAt(self, row, col, binTrans, iTarget=None, isSelected=False):
         """ Set the rail at cell (row, col) to have transitions binTrans.
             The target argument can contain the index of the agent to indicate
             that agent's target is at that cell, so that a station can be
@@ -76,7 +76,7 @@ class GraphicsLayer(object):
         """
         pass
 
-    def setAgentAt(self, iAgent, row, col, iDirIn, iDirOut,isSelected=False):
+    def setAgentAt(self, iAgent, row, col, iDirIn, iDirOut, isSelected=False):
         pass
 
     def resize(self, env):
diff --git a/flatland/utils/rendertools.py b/flatland/utils/rendertools.py
index d7dca5d7e3a66dc95fca60fe3886b1454086a498..52892478b9dcea437e5f176af1db05cf4ad66f2f 100644
--- a/flatland/utils/rendertools.py
+++ b/flatland/utils/rendertools.py
@@ -762,8 +762,7 @@ class RenderTool(object):
                 if agent is None:
                     continue
                 dTargets[tuple(agent.target)] = iAgent
-                dSelected[tuple(agent.target)] = (iAgent==iSelectedAgent)
-
+                dSelected[tuple(agent.target)] = (iAgent == iSelectedAgent)
 
             # Draw each cell independently
             for r in range(env.height):
@@ -777,7 +776,7 @@ class RenderTool(object):
                         target = None
                         isSelected = False
 
-                    self.gl.setRailAt(r, c, binTrans, iTarget=target,isSelected=isSelected)
+                    self.gl.setRailAt(r, c, binTrans, iTarget=target, isSelected=isSelected)
 
         for iAgent, agent in enumerate(self.env.agents):
 
@@ -795,7 +794,7 @@ class RenderTool(object):
 
             # setAgentAt uses the agent index for the color
             # cmap = self.gl.get_cmap('hsv', lut=max(len(self.env.agents), len(self.env.agents_static) + 1))
-            self.gl.setAgentAt(iAgent, *position, old_direction, direction, iSelectedAgent==iAgent)  # ,color=cmap(iAgent))
+            self.gl.setAgentAt(iAgent, *position, old_direction, direction, iSelectedAgent == iAgent)
 
         if show_observations:
             self.renderObs(range(env.get_num_agents()), env.dev_obs_dict)