Updated comments to clarify flow