HW4 Added loop
HW 4 Assignment