Update one_off.py