added model diagram