Update prune_files.py