Adding more funtionality to the main menu - more working buttons