just a minor change in controller.js