Update markov_chain_simulation.py