Change sort method a bit