add diagram + aes