more docstrings
more tests