Update TreeDepth.java