Update sort_files.py