Improved Diffusion Main