numba decorator modified with 2 additional arrays