Updated the docs to reflect the new python code