Improved the menu screen