Update compare.py