python make sqrt with while