add docstrings
update docstrings to be clearer