Update DiagramService.java