cleaned up 3D diffusion solver and commented code