Updated diagrams and README.md

As per title