Update tree_kernel.py