Improve doc string