Make model save on add/remove