Add flatten_collection