Add a Unflatten component (opposite of Flatten)