Updated collection to support array of documents