update docstrings and readme