DOC: improved doc string