DOC : minor doc-string changes
MNT : simplified if staments