Update Diffusion.h