Add AstExtension and implement AstMap correctly.