Add comparator class