Added a few minor comments

Wanted to check the code, to gain insight on the work that I can do to improve, and hence while going through the code, I added a few comments.