finished updating and testing prune_model