Major updates. Working water shaders