class diagram updates