updated the class diagram