Now that we removed `CommentStatement` from the AST as nodes, we need a way to discover all comments when walking the AST. Maybe a new WalkMode option that includes comments?