Organized the code in functions
Upload simplified UML diagram
New images