Update sort.py