Update comp.py