Update pa1.cpp