second version of linear diffusion for module2