Update sort.py

fixed formatting