Added main menu code