some diagram changes