Cleaned up category UI