Designing the model class