updating unit tests and documentation