Added more of the shader