updating class diagram