Added diffusion fluxes to the viscous flux calculations