Small edit to MCMC steps