updated DiagramTest