Update uml for ast