Update part 1.cpp