updated AST example in docs