Added functionality to the main menu