minor edits to ode.py