Minor update to the comments