Update optagent.py