Docstring fixes as per review