Update HW4_V6.cpp