Update lab4.cc