Started work on the tree_utils.py file.