Add square code