Added a few comments to discuss the new code