[Anton] Working on the main menu