Update HW4_v3.cpp