Added test, and removed __array__