Update sorted_data.py

itemgetter is my friend