Add some functionality to the main menu