Update lab3.cpp

Part 2