First version add diagram