Adding the model classes