doc string improvement