refactoring code di controller.js