Improving sorting UI