Modifyed controller.js