Updated the comments to reflect the new version