Rewriting the animation with AnimatedWidget