First order ODE in Python