Adding stuff to the menu function: