Update sort.py

something