Added main menu update code