Adding success_rate() logic