Update sort_data.py