Update batch_run.py