Small diagram changes