Solving Ode Using Python Help