First draft of 2d diffusion