Added some changes to the menu and the UI.