Updates to make Jonas' code run.

The ODE code in the test case runs successfully. I'm not sure those results are right as I'm not sure what it's doing.