Add additional test case; add gradient descent fallback