Add Retry class