Moved the check. I think `addClass` is pretty obvious.