Add a confidence threshold (0.5)