Set batch size
Merge pull request #19 from hakobera/set-batch-size

Set batch size