Add clamp function