Update TreeSetTest.java