Update A1-code.cpp