added documentation for batch_size argument