Address review comments: Slightly better docs.