Add square feature