Update lab.cpp