Setup for read_batch mode a little differently: