Square Root Function Python Np