Update square.py
Merge pull request #1 from Migwee/patch-1

Update square.py