Add retry count paramter