Perfect Square Function In Python W3schools