Got rid of a special case in the transition matrix computations