Update lab3.cpp