Update part2.cpp