adapt class diagram after refactoring