Cleaned up main menu UI