Change comments based on node type