updated model class