Improve 3d Diffusion Equation