include NumPy integration