Update final_sorting.py