Update main.js

ng-class changes due to controller