add diagram stuff 2