Wire up ToolBelt#groups

Added ToolBelt#query for simpler filtering of AST nodes
More tests