add batch_size option