adding more to sort method