added viscosity into momentum.cpp