cleaning up
playing around with Numba
more numba