fixes to class diagram