Fixed bug with maze.py