Adding the maze code