Work on main menu (WIP)