Add comments to README code