Make category display smarter