Button class now changes with ng-class