documentation for Diffusion2D