Added nonlinear diffusion module