Update heap.py