Enable saving the state of MCMC