Work on ODE exercise