Updated doc-comments to reflect changes