Update batch.py