updated model diagram