Update run_batch.py