Improve UX for sorting