Adds some comments to the code