Add document_id to documents