Update square method
Fix typo