Doc string improvements