Added model diagram