Added functionality to the menu