numba is awesome
serious mistake fixed :S