Update TreeSetDemo.java