Added comments to make changes more obvious