Alter the class for model