Updated MainController