First maze.py prototype