Add model class