cleanup for problem 4