Made some changes to the comments