Adding viscosity loop to convert to opencl