Updating class diagram