Refactoring on the shader.