Rename variable category_type -> node_type
Add TODO