Working single level twodtree.py with test script