Add filter option to numba script