Added the diffusion task