improve categories UI