Added explicit non-linear diffusion.