updated diagram for ast