Added the necessary class from the class diagram.