fix diagrams
Merge pull request #2 from anilsaldhana/master

see comments