Add argument to specify diagram type
Update README.md
Clean up
Merge branch 'feature/DiagramType' into develop