changed treeset example