[ENH] Diffusion in 2D ( issue #1 )