Class diagram update