Add a couple comments to the code