HW4 code review added