Could be done in numba as a prototype
Could be done in numba as a prototype