Changed the menu to be just a button