Cont to implement the diffusion test for 2D