Add some comments to the code