Add a clamp function