Update the `diagrams` tool to use a category as root entity