add return & call njit