Class diagram updates