fixing HW#4

fixing HW#4