Changed the logic for heapify
Updated tests for keko