Update docstring according to @stefanv's comments