Functionality to main menu and end-game screen