Perfect Square Function In Python Example