Update square.h