Minor typo fix and adding comments