Improved menu
Small fix