Higher Order Markov Chain Python