Update TransitionFigure.java