Updates to the menu function