Update sort.py

changed the args number