Minor updates to diagram, screenshot added