Update markov_chains.py

added import statements