Square Root In Python Function