Added classes to use in class diagram