Add animation on height changes