Shader clean up