rewrite controller.js