Add pxd file for tree in setup.py