Replaced mistaken HW3 code to HW4 code in the .cpp file