the linear part works correctly, testing the nonlinear now