add diagram stuff 3