refactor($animate): use $animate.$$setClassImmediately to save code

This is not fixing anything, it's just re-organizing and removing comments indicating that a proper fix is needed. I don't think we should merge this. What we should do is figure out how to fix it properly. I've offered suggestions, but you know the animation code a lot more than I do.