Add Heap code