further updates to class diagram