unit tests & update docs, uml