Added the model classes