Added Uml diagram