Create heap.py