Complete TestDiagramScene class