Add support for category nodes