Refactor sort benchmark script