Minor code cleanup, added Keyword arguments from ODE.jl API