Started working on UML diagram