Revert cast, improved query function

Cast was not needed since the return value was already being cast. Query
function changed to use heapq.nlargest() for simplicity