Small updates to the comments