Updated python example code sort.py