Improved doc string