@astrofrog - OK, now is:
```
import astropy.units as u, numpy as np
```