Revised the driver to start creating trees using the tree builder