Add comparator.py