Update invariant.py