add support for categories