Fixes to MCMC notebook