Update lab4_final.cc