More sorting, don't understand why it works on word_count4.py