HW3 part 3 CleanUp