Update docs for slightly nicer API.