Easy mode Simple Sorting Python2 solution