added a few comments to the code
Added comments to the code to finish up