Minor informational update

* Improving UML file

* Adding comments