Fix docstring (issue #134) for jit functions