Tests cleanup. Add batch_amount parameter to the batch_add method