Updated the comments

Updated comments and README.