HW class 4 exercise 2 cleanup