rewrite controller.js
change