Simulating an ordinary differential equation with SciPy