Added code for main menu