** ADD Flattening Function to Output Element-Wise Results **