Added a model class