Minor changes (added comments to clarify new code)