Adding tests for comparator