output as an np.array and not as list, much bettah: v1.2.0