Update controller.js