update class diagram docs
Merge pull request #125 from camallen/update_class_diagram

update class diagram docs