HW_3 sorting.py