Also updated the png diagram