Allow batch size to be overridden