Added a couple of class diagrams