Update sort_engine_output.py