Update three_way_quicksort.py