ENH: add cleanup tool