Update Q_2_1.cpp