Update to predict all observations in one pass through the tree
Tidy up for the improvement to the speed of predicting