Update batch_imaging.py