Updated UI in the menu