Add 4 lines of cleanup

Final commit for today.