Moved main menu to separate class