Update API documentation to use proper docstrings

Change-Id: I24bc5c60204b750259f1866b640b41ecbb4d5b20