x0 = np.deg2rad([5.0, -3.0, 0.0, 0.0])

model.plot_simulation(times, x0,
    input_func=lambda t, x: np.array([0.0, 50.0*x[2]]),
    v=2.0)