Better docstring and README