Make batch_set() more flexible to accept initial data available during the batch process