ENH: cleanup