improve ng-class