Python Keep Rows Together When Sorting Files