Class diagram update

Signed-off-by: Benoit Brayer <brayer.benoit@gmail.com>