edit controller.js