Add batch_size functionality to spaces + test.