Update unit tests and add docstring comments