Abstracted out the model class