Added square method