modify controller.js