python fsolve with arguments