Update docstrings and README