small optimizations and cleanup for viscosity/vorticitiy