add additional methods to model diagrams