Add diffusion.m