Modified imports to work with new file structure.
Implemented regression tree.