Finished working sort.py