Almost Done with problem 4 in HW1