Changes in the comments, to be consistent with model docs