cleanup problem 4 #2