MainController updated