Updated comments to clarify recent changes in code