Modified mainController.js