Updated documentation to reflect changes to the .py file