Minor update to cleanup script.