Happy with this approach, and have renamed models to be trees.py