Update README
Better docstrings