Made changes to include the citation for the diagram