still figuring out an efficient way to represent the transition probability matrix