Add num_docs to categories piece