make numba optional
cleanup whitespace and weird docs