Update heaps.py