Modify vertex shaders to use new uniforms