Updated Diffusion docs for 2.0