Update lab_2.cpp