Seems to be working. need to add comparator in models.py