I updated the main function by putting the condition for the new map.