Update docs
Update doc string to generate API docs