Update replay_sort.py