Added squaring functionality