Added functionality to main menu