Added some comments to new code