Update lab2.cpp