Update some comments to reflect changes