work on numba and NumPy notebook