Update mcmc.py