modify selection_sort.py