sig2_epsilon

You did the correction by adding another variable sig2_epsilon. Plus
other modifications to simulation, gibbs, and diag.