Updated the docs to reflect the code changes