Update code snips for new viscosity inputs