Adding Flatten()