Started to add some uml diagrams