Add TreeSet class