update init_transition_matrix
Add code to initialize transition matrix