Update controller.js

This will fix Issue #5