add kth_bst.py