plotting stuff in DiagonalGaussian