If you had defined generalized speed variables you could insert them here instead of the derivatives.

```python
A.set_ang_vel(I, u[0] * I.z + u[1] * I.x)
```

Also did you set the angular velocity of I equal to zero. I think you need to, but not sure.