add diagram visualisation