Expand 2d Numpy Array To 3d