Added some changes in the comments