added some things in the UML