Updated

* Adding two class in the UML
Coding them ASAP