Update code4.cpp