Update hw6_main.cpp