Add list comprehension to quick_sort
shrunk pivot function