update unit tests, uml and docs