Diffusion for particles count works in 2d sample