Class diagram changes