edit to class diagram