Create kth_element.py