add missing import heapq