Update docstring and documentation
Quality and unittests improvements