Update HW2.cpp