Make batch size configurable.