Water shader improvements