Working on get_trans_prob function