model diagram
adjust model classes