adding new unit tests and uml