Add some comments to new code