Adding support for micro batching