Updated the class diagram