Add model save feature.