Clean up of class diagram
Merge pull request #8 from Postmodum37/master

Clean up of class diagram