Added model class