Updated: the controller.js
Updated: the readme (live demo)