Minor documentation improvements

Minor documentation improvements I did while reading the code.