Working on the main menu screen