Update comments to reflect new implementation