Add more elements to DiagrammController