Small NDArray improvements