Update comments to reflect current set up