Adding 1D and 2D Diffusion