Updated version of the class diagram