Add class_tree.py to repo