updated the map to seperate into individual stages