Update hw1q4.cpp