modified controller.js
modify controller.js