Update selection_sort.py