Using arrays in Numba#