work on docstring and documentation