Updated controller.js