Added saturation of diffuse method back in