updates to class diagram (added notes and fixed some associations)