Update to use ng-animate