modifying the cleanup script