Updated For HW4-Part2