small update to prune.m