improved readme and docstring