Add the model classes