Update comments to include type param