Addded classes to the model