Add benchmark_sort function