add diagram stuff