Python Square Root Function