add/fix/improve docstrings