Update heapify.py