Update prim_maze_gui.py