Use partial_sort