Add support for batching + setting on schemas to be marshaled