Return to Diffusion_try2