Updated Maze Generation code