Updated functions to run mcmc and plot results