Update sort.py

Removing test code