modified controller.js