adding some diagram stuff