Updated code for plots
Updated code for running mcmc and plots