additional diagram changes