Higher Order Markov Chain Python Code