slight diagram changes