Square Root Formula In Python