Update theory.py