small change to class diagram