diff --git a/robotgame/block.py b/robotgame/block.py
index 15df8f3..bea08be 100644
--- a/robotgame/block.py
+++ b/robotgame/block.py
@@ -197,8 +197,7 @@ class ActionBlock(Block):
     def __init__(self, level, x, y, action=None, movable=True):
         self.__dict__.update(locals())
         Block.__init__(
-            self, level, x, y, img=level.imgs['block1'],
-            movable=movable)
+            self, level, x, y, 'block1', movable=movable)
         self._last_pos = None
 
     def update(self, e, t, dt):
diff --git a/robotgame/level1.py b/robotgame/level1.py
index 23744ba..1928e17 100644
--- a/robotgame/level1.py
+++ b/robotgame/level1.py
@@ -203,7 +203,7 @@ class Level1(level.Level):
         boxes += boxes
         boxes += [logic.colourboxes.generate_random_box(1) for _ in range(3)]
         random.shuffle(boxes)
-        
+
         pos_colour = {}
         for box, (x, y) in zip(boxes, itertools.product(range(3), range(3))):
             # self.tiles.append(tile.Tile(self, x * 64 + task3_pos[0],
@@ -216,13 +216,15 @@ class Level1(level.Level):
                          for i in range(3)]
         self.objects.extend(action_blocks)
 
-        wells = [block.ColorWell(self, task3_pos[0] + 64, task3_pos[1] + 48 * 5)]
+        wells = [block.ColorWell(self, task3_pos[0] + 64,
+                                 task3_pos[1] + 48 * 5)]
         self.objects.extend(wells)
 
         def update_wells(block):
             cur_boxes = []
             for block in action_blocks:
-                box = pos_colour.get(((block.x - task3_pos[0]) / 64, (block.y - task3_pos[1] - 48) / 48))
+                box = pos_colour.get(((block.x - task3_pos[0]) / 64,
+                                      (block.y - task3_pos[1] - 48) / 48))
                 if box:
                     cur_boxes.append(box)
 
@@ -235,8 +237,7 @@ class Level1(level.Level):
 
         for b in action_blocks:
             b.action = update_wells
-        
-        
+
         ### Task 4: Inverted bits
 
         task4_pos = (64 * 13, 48 * 13)