Added batchSize parameter