Updated comments to reflect changes