make batch size configurable