Small changes to shaders