Update NBTree.py