Make AnimatedTransition API simpler