Update maze_gen.py