[Update] Model classes