Updated for tree splits