Add the comments

Added the comments and made small changes