UI improvements (menu)