improve controller.js