Small update to shaders