updated cleanup script a little bit