Adding AST parsing