Changes to the menu (this isn't very good atm, will work on)