Improve category display