Finished the diffusion task