Added "square" method