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

I just explained above why I changed this around. So please shed some light as to why you think this is wrong?