Changing [].flatten to array wrapper method