Minor changes to the comments in the code