Improve docstring for shape