Adding more to the TransitionFunction class