Clean up AST creation