Added callback to the animation