update readme and docstring