Model augmentation