10 factor speedup by directly creating 3D array instead of np.array(list of array)