Update hw2.cpp