fix diagrams
Merge pull request #118 from anilsaldhana/master

see comments