Update HW02.cpp