Update lab_2.cc