add missed model class