added classes to the model