Add base_tree to input example