Save status before modifying the viscous source term
Added diffusion terms to the viscous source terms