Added in main menu button functionality