Added recalculation for viscosity