Minor changes to the HoeffdingTree code