Add mask to the shader