Added a few comments to the code