more diagram changes