updated diagram with new models