batch_size needs to be here as a kwarg now
Merge pull request #41 from majidaldo/batchsize

batch_size needs to be here as a kwarg now