Update optimusPrime.py