Improve UI for category page