updated controller.js