Added the class diagram. :)