Added saturation control