Updated import_treetm.py