Update classtree.py