Update the comments to reflect the new changes