Fix bug

week2/solution/agent_programs.py

@@ -250,7 +250,7 @@ def utility_function(model, cur_location, direction):
             break
     
     # check if there is dirt in that direction
-    for cell in cells:
+    for cell in filtered_cells:
         if cell == 'D':
             # there is dirt, return high utility
             return 999