Small reorganization in the water shader
Speedy
Might be good to use the values