Update comments to reflect class name changes.