Update custom-controller.js