Add ASTFlatten