Added the heap in Python