Updated animation methods to pass the transition context too.