Add --prune option