Class diagram change