python implementation of heap_sort v2