Adding a model class