Minor updates on the docs and function name