Add separate AST stage