Added new functionality for category UI display