Update to class diagram