added the class diagram