class diagram updates
class diagram updates