Adding docstring and updating readme to match new API.