Add solver for diffusion equation