Added code for HW 4 part 1.