Fix and improve doc string