Update square.cpp