simplify heap_sort.py .