Create prim_maze.py