modify & add diagram image