updated comments to reflect changes in code