Improve Diffusion