add diagram code