oops forgot to add the new Model class