Updated comments to better reflect the updated script