major modifications to TransitionMatrix.py to include initial state distribution