Python Take Square Root