Math Square Root Function Python