Square function fails tests

[Sumukh]

The square functions fails its tests!!!!!

[jerryhchen97]

I'll work on this

[duncaninganji]

I want to work on this

[springle]

I submitted a pull request that should resolve this issue.

[JohannBanta]

I'll work on this

[XinchenXu]

I've resolved this issue.

[lingjieqiao]

I will work on this dadadda