Scipy optimize root