Update TreeHeight.java