Fixed bug for viscosity. It should work now