1. Add Retry logic